package com.arithmetic.basics;

/**
 * 给定一个二叉树，判断它是否是 平衡二叉树
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * 输入：root = [3,9,20,null,null,15,7]
 * 输出：true
 * 示例 2：
 * <p>
 * <p>
 * 输入：root = [1,2,2,3,3,null,null,4,4]
 * 输出：false
 * 示例 3：
 * <p>
 * 输入：root = []
 * 输出：true
 * 提示：
 * <p>
 * 树中的节点数在范围 [0, 5000] 内
 * -104 <= Node.val <= 104
 * Related Topics
 */
public class Day31 {
    /**
     * 递归
     *
     * @param root
     * @return 这段代码用于判断二叉树是否为平衡二叉树。
     * isBalanced方法递归检查每个节点的左右子树高度差是否不超过1，
     * 并且左右子树本身也是平衡的。getHeight方法递归计算二叉树的高度
     */
    public boolean isBalanced(Day28.TreeNode root) {
        if (root == null) {
            return true;
        }
        if (Math.abs(getHeight(root.left) - getHeight(root.right)) > 1) {
            return false;
        }
        if (isBalanced(root.left) && isBalanced(root.right)) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * 获取二叉树的高度
     *
     * @param root
     * @return
     */
    public int getHeight(Day28.TreeNode root) {
        if (root == null) {
            return 0;
        }
        return Math.max(getHeight(root.left), getHeight(root.right)) + 1;
    }
}
